package com.woniuxy.util;

import com.rabbitmq.client.Channel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;

import java.io.IOException;
import java.util.function.Consumer;
import java.util.function.Supplier;

/**
 * RabbitMQ消息处理工具类，提供消息确认和异常处理功能
 */
@Slf4j
public class RabbitMQMessageHelper {
    /**
     * 执行消息处理并自动处理ACK/NACK
     * @param channel RabbitMQ通道
     * @param message 消息对象
     * @param action 要执行的业务逻辑
     */
    public static void processMessageWithAck(Channel channel, Message message, Consumer<Message> action) {
        try {
            // 执行核心业务逻辑
            action.accept(message);
            // 确认消息
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
        } catch (Exception e) {
            log.error("消息处理失败，准备拒绝消息: {}", e.getMessage(), e);
            try {
                // 拒绝消息并不重新入队（避免死循环）
                channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, false);
            } catch (IOException ex) {
                log.error("消息拒绝失败: {}", ex.getMessage(), ex);
            }
        }
    }

    /**
     * 执行消息处理并自动处理ACK/NACK，支持返回结果
     * @param channel RabbitMQ通道
     * @param message 消息对象
     * @param supplier 要执行的业务逻辑，带返回值
     * @param <T> 返回值类型
     * @return 业务逻辑执行结果
     */
    public static <T> T processMessageWithAck(Channel channel, Message message, Supplier<T> supplier) {
        try {
            // 执行核心业务逻辑
            T result = supplier.get();
            // 确认消息
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
            return result;
        } catch (Exception e) {
            log.error("消息处理失败，准备拒绝消息: {}", e.getMessage(), e);
            try {
                // 拒绝消息并不重新入队（避免死循环）
                channel.basicNack(message.getMessageProperties().getDeliveryTag(), false, false);
            } catch (IOException ex) {
                log.error("消息拒绝失败: {}", ex.getMessage(), ex);
            }
            return null;
        }
    }
}    